#ifndef SAVEIMAGEDIALOG_H
#define SAVEIMAGEDIALOG_H

#include <QImage>
#include <QDialog>
#include <QPushButton>
#include "imagewidget.h"

namespace views
{
    class SaveImageDialog: public QDialog
    {
        Q_OBJECT

    public:
        SaveImageDialog(QWidget* parent);
        SaveImageDialog(QWidget* parent, QImage& image);
        virtual ~SaveImageDialog();
        void    initUI();
        void    setupUI();
        void    setImage(QImage& image);

    public slots:
        void saveImage();
        void cancelSave();

    private:
        QImage      *_image;
        QPushButton *_btnSave;
        QPushButton *_btnCancel;
        ImageWidget *_imageWidget;
    };
}

#endif // SAVEIMAGEDIALOG_H
